package log

import (
	"context"
)

const _KEY_TRACE_ID string = "trace_id"

func AddTraceID(ctx context.Context, traceID string) context.Context {
	return context.WithValue(ctx, _KEY_TRACE_ID, traceID)
}

func GetTraceID(ctx context.Context) string {
	id, ok := ctx.Value(_KEY_TRACE_ID).(string)
	if !ok {
		return ""
	}
	return id
}
